#!/bin/bash

# display: realA fakeB fakelrB recA idtB  
# dataset:BDD_s2r SYN_d2n GTA2CITY
python train.py --gpu_ids 0 --display_id 12 --display_ncols 5 \
--dataroot /data1/2021/lyf/datasets/SYN_d2n --dataset_mode unaligned \
--name SYN_d2n_3stage --model swinunit --netG swin_unit_nstage --stage_num 3 --netD nstage --batch_size 3 \
--display_freq 500 --checkpoints_dir ./checkpoints/nstage

# python test.py --gpu_ids 1 \
# --dataroot /data3/2021/lyf/datasets/BDD_s2r --dataset_mode unaligned \
# --name BDD_s2r_3stage --model swinunit --netG swin_unit_nstage --stage_num 3 --netD hrlr \
# --preprocess resize --resize_height 768 --resize_width 1280 \
# --epoch_count 100 --checkpoints_dir ./checkpoints/ --phase test --results_dir ./results/nstage
